package com.dh.leetcode.seven;

import org.junit.Assert;
import org.junit.Test;

/**
 * @ClassName: _766_toeplitz_matrix
 * @Description: 766. 托普利茨矩阵
 * 给你一个 m x n 的矩阵 matrix 。如果这个矩阵是托普利茨矩阵，返回 true ；否则，返回 false 。
 * <p>
 * 如果矩阵上每一条由左上到右下的对角线上的元素都相同，那么这个矩阵是 托普利茨矩阵 。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * 输入：matrix =
 * [[1,2,3,4],
 * [5,1,2,3],
 * [9,5,1,2]]
 * 输出：true
 * 解释：
 * 在上述矩阵中, 其对角线为:
 * "[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]"。
 * 各条对角线上的所有元素均相同, 因此答案是 True 。
 * 示例 2：
 * <p>
 * <p>
 * 输入：matrix = [[1,2],[2,2]]
 * 输出：false
 * 解释：
 * 对角线 "[1, 2]" 上的元素不同。
 * <p>
 * https://leetcode-cn.com/problems/toeplitz-matrix/
 * @Author: shouzimu
 * @Date: 2021/2/22 10:03
 */
public class _766_toeplitz_matrix {

    /**
     * 遍历每个元素是否和左上角元素相等
     * <p>
     * 左上角元素即为 (i-1)(j-1)
     *
     * @param matrix
     * @return
     */
    public boolean isToeplitzMatrix(int[][] matrix) {
        int x = matrix.length, y = matrix[0].length;
        for (int i = 1; i < x; i++) {
            for (int j = 1; j < y; j++) {
                if (matrix[i][j] != matrix[i - 1][j - 1]) {
                    return false;
                }
            }
        }
        return true;
    }

    /**
     * 左右两个角不对比
     * 代码比较繁琐
     *
     * @param matrix
     * @return
     */
    public boolean isToeplitzMatrix_old(int[][] matrix) {
        int x = matrix.length;
        int y = matrix[0].length;
        // i=y-1=>0;j=0
        // 左下部分
        for (int i = x - 2; i >= 0; i--) {
            //查找该对角是否都相等
            int v = matrix[i][0];
            int k = 1;
            for (int j = i + 1; j < x && k < y; j++) {
                if (matrix[j][k] != v) {
                    return false;
                }
                k++;
            }
        }
        // i=1;j=0=>y-1
        // 右上部分
        for (int i = 1; i < y - 1; i++) {
            //查找该对角是否都相等
            int v = matrix[0][i];
            int k = 1;
            for (int j = i + 1; j < y && k < x; j++) {
                if (matrix[k][j] != v) {
                    return false;
                }
                k++;
            }
        }
        return true;
    }

    @Test
    public void isToeplitzMatrixTest() {
        Assert.assertEquals(true, isToeplitzMatrix(new int[][]{{65, 98, 57}}));
        Assert.assertEquals(true, isToeplitzMatrix(new int[][]{{1}}));
        Assert.assertEquals(true, isToeplitzMatrix(new int[][]{{1, 2, 3, 4}, {5, 1, 2, 3}, {9, 5, 1, 2}}));
        Assert.assertEquals(false, isToeplitzMatrix(new int[][]{{1, 2}, {2, 2}}));
    }
}
